package com.taldh.algorithm.array;

import org.junit.Assert;
import org.junit.Test;

import java.util.Arrays;
import java.util.HashMap;
import java.util.stream.Collectors;

public class TwoSumSolution {

    /**
     * 001. 两数之和
     * 给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。
     *
     * 你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。
     *
     * 示例:
     *
     * 给定 nums = [2, 7, 11, 15], target = 9
     *
     * 因为 nums[0] + nums[1] = 2 + 7 = 9
     * 所以返回 [0, 1]
     *
     * 来源：力扣（LeetCode）
     * 链接：https://leetcode-cn.com/problems/two-sum
     * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
     *
     * @param nums
     * @param target
     * @return
     */
    public int[] twoSum(int[] nums, int target) {
        //暴力解法，时间复杂度为n^2, 空间复杂度 1
//        for (int i = 0; i < nums.length-1; i++) {
//            for (int j = i+1; j < nums.length; j++) {
//                if (nums[i]+nums[j] == target) {
//                    return new int[]{i,j};
//                }
//            }
//        }
//        return null;
        //hashmap打平  时间复杂度为n, 空间复杂度 n
//        HashMap<Integer, Integer> map = new HashMap<>(nums.length);
//        for (int i = 0; i < nums.length; i++) {
//            map.put(nums[i], i);
//        }
//        for (int i = 0; i < nums.length; i++) {
//            int a = nums[i];
//            Integer b;
//            if ((b = map.get(target - a)) != null && i != b) {
//                return new int[]{i,b};
//            }
//        }
//        return null;
        //hashmap打平同时检查    时间复杂度为n, 空间复杂度 n

//        HashMap<Integer, Integer> map = new HashMap<>(nums.length);
//        for (int i = 0; i < nums.length; i++) {
//            int a = nums[i];
//            Integer b;
//            if ((b = map.get(target - a)) != null && i != b) {
//                return new int[]{i,b};
//            }
//            map.put(nums[i], i);
//        }
//        return null;

        //究极空间换时间
        int volume =2048;       //100000000000
        int bitMode = volume-1;//011111111111
        int [] result =new int[volume];

        for (int i=0;i<nums.length;i++){
            int c = (target - nums[i]) & bitMode;
            if (result[c]!=0){
                return new int[]{result[c]-1,i};
            }
            result[nums[i] & bitMode]=i+1;
        }
        return null;
    }

    /**
     * 167. 两数之和 II - 输入有序数组
     * 给定一个已按照升序排列 的有序数组，找到两个数使得它们相加之和等于目标数。
     *
     * 函数应该返回这两个下标值 index1 和 index2，其中 index1 必须小于 index2。
     *
     * 说明:
     *
     * 返回的下标值（index1 和 index2）不是从零开始的。
     * 你可以假设每个输入只对应唯一的答案，而且你不可以重复使用相同的元素。
     * 示例:
     *
     * 输入: numbers = [2, 7, 11, 15], target = 9
     * 输出: [1,2]
     * 解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。
     *
     * 来源：力扣（LeetCode）
     * 链接：https://leetcode-cn.com/problems/two-sum-ii-input-array-is-sorted
     * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
     */
    public int[] twoSum2(int[] numbers, int target) {
        if (numbers == null || numbers.length == 0) {
            return null;
        }

        int i = 0;
        int j = numbers.length - 1;
        while (i < j) {
            if ((numbers[i] + numbers[j]) == target) {
                return new int[]{i+1,j+1};
            } else if ((numbers[i] + numbers[j]) > target) {
                j--;
            } else {
                i++;
            }
        }
        return null;
    }

    @Test
    public void twoSumTest() {
        int[] nums = {3,3};
        int target = 6;
        Assert.assertArrayEquals(twoSum(nums, target), new int[]{0,1});
    }

    @Test
    public void twoSum2Test() {
        int[] nums = {2, 7, 11, 15};
        int target = 9;
        Assert.assertArrayEquals(twoSum2(nums, target), new int[]{1,2});
    }
}